package org.jpro.datasource;

import org.jpro.enums.DataSourceType;

/**
 * <P>功能描述 : 获得和设置上下文环境 主要负责改变上下文数据源的名称 </P>
 * <P>File name : DataSourceContextHolder </P>
 * <P>Author : 王泽浩 </P>
 * <P>E-Mail : 1028625100@qq.com </P>
 * <P>Date : 2017/1/1 下午8:59</P>
 */
public class DataSourceContextHolder {

    // 线程本地环境
    private static final ThreadLocal<DataSourceType> THREAD_LOCAL = new ThreadLocal<DataSourceType>();

    // 获取数据源类型
    public static DataSourceType getDataSourceType() {
        return THREAD_LOCAL.get();
    }

    // 设置数据源类型
    public static void setDataSourceType(DataSourceType dataSourceType) {
        THREAD_LOCAL.set(dataSourceType);
    }

    // 清除数据源类型
    public static void clearDataSourceType() {
        THREAD_LOCAL.remove();
    }

}
